package com.zjsystem.config.mybatisPlus;

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.GlobalConfig;
import com.baomidou.mybatisplus.generator.config.PackageConfig;
import com.baomidou.mybatisplus.generator.config.StrategyConfig;
import com.baomidou.mybatisplus.generator.config.po.TableFill;
import com.baomidou.mybatisplus.generator.config.querys.DMQuery;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;

import java.util.ArrayList;

public class MpCode {
    public static void main(String[] args) {
        //需要构建一个代码生成器对象
        AutoGenerator autoGenerator = new AutoGenerator();
        //配置策略
        //1.全局配置
        GlobalConfig gc = new GlobalConfig();
        String projectPath = System.getProperty("user.dir");//获取项目根路径
        System.out.println(projectPath);
        gc.setOutputDir(projectPath+"/src/main/java");//代码输出的路径
        gc.setAuthor("wyf");//作者
        gc.setOpen(false);//是否打开win文件夹
        gc.setFileOverride(true);//是否覆盖原有的文件
        gc.setServiceName("%sService");//去掉Service前面的I
        gc.setIdType(IdType.AUTO);//设置自增id策略
        gc.setDateType(DateType.ONLY_DATE);//设置默认的日期时间
//        gc.setSwagger2(true);//生成swagger文档
        autoGenerator.setGlobalConfig(gc);

        //2.设置数据源配置
        DataSourceConfig dsc = new DataSourceConfig();
        dsc.setDbType(DbType.MYSQL);//数据库类型
        dsc.setUsername("root");//用户名
        dsc.setPassword("123456");//密码
//        dsc.setDriverName("dm.jdbc.driver.DmDriver");//数据库驱动
        dsc.setDriverName("com.mysql.cj.jdbc.Driver");//数据库驱动
        dsc.setUrl("jdbc:mysql://10.54.148.17:3306/DM?useSSL=false&useUnicode=true&characterEncodiing=utf-8&serverTimezone=GMT%2B8");//地址
//        dsc.setUrl("jdbc:dm://10.65.50.17:5236/RZXDKZ");//地址
//        dsc.setDbQuery(new DMQuery());
        autoGenerator.setDataSource(dsc);

        //3.包的配置
        PackageConfig pc = new PackageConfig();
        pc.setModuleName("zjsy");//模块名称
        pc.setParent("com.zjsystem");//父包
        pc.setEntity("dto");//实体类的包
        pc.setMapper("mapper");//mapper的包
        pc.setService("service");//service的包
        pc.setController("controller");//controller的包
        autoGenerator.setPackageInfo(pc);

        //4.策略配置
        StrategyConfig strategy = new StrategyConfig();
        strategy.setInclude("chat_messages");//设置要映射的表名
        strategy.setNaming(NamingStrategy.underline_to_camel);//属性名_转驼峰命名
        strategy.setColumnNaming(NamingStrategy.underline_to_camel);//列名_转驼峰命名
        strategy.setEntityLombokModel(true);//自动生成lombok
        strategy.setRestControllerStyle(true);//自动生成Rest风格
        strategy.setLogicDeleteFieldName("deleted");//设置逻辑删除
        strategy.setVersionFieldName("version");//设置乐观锁
        //自动填充策略
        TableFill createtime = new TableFill("create_time", FieldFill.INSERT);//创建时间
        TableFill updatetime = new TableFill("update_time", FieldFill.INSERT_UPDATE);//修改时间
        ArrayList<TableFill> tableFills = new ArrayList<>();
        tableFills.add(createtime);
        tableFills.add(updatetime);
        strategy.setTableFillList(tableFills);//设置自动填充
        strategy.setControllerMappingHyphenStyle(true);//设置请求路径格式 例：localhost:8080/hello_id_2
        autoGenerator.setStrategy(strategy);

        //5.执行代码生成器
        autoGenerator.execute();

    }
}
